import java.util.HashMap;
import java.util.Map;

class Solution4 {
    public int subarraysDivByK(int[] nums, int k) {
        int sum = 0, n = nums.length, ret = 0;
        Map<Integer, Integer> map = new HashMap<>();
        // 数组中的全部元素之和 % k == 0；处理边界情况
        map.put(0, 1);
        for(int x : nums) {
            sum += x;
            int tmp = (sum % k + k) % k;
            if(map.containsKey(tmp)) {
                ret += map.get(tmp);
            }
            map.put(tmp, map.getOrDefault(tmp, 0) + 1);
        }
        return ret;
    }
}